/**
 * 有 n 个城市，其中一些彼此相连，另一些没有相连。如果城市 a 与城市 b 直接相连，且城市 b 与城市 c 直接相连，那么城市 a 与城市 c 间接相连。
 *
 * 省份 是一组直接或间接相连的城市，组内不含其他没有相连的城市。
 *
 * 给你一个 n x n 的矩阵 isConnected ，其中 isConnected[i][j] = 1 表示第 i 个城市和第 j 个城市直接相连，而 isConnected[i][j] = 0 表示二者不直接相连。
 *
 * 返回矩阵中 省份 的数量。
 *
 *https://leetcode.cn/problems/number-of-provinces/description/
 */
class FindCircleNum {
    public int findCircleNum(int[][] isConnected) {
        int n = isConnected.length;
        UnionFindSet unionFindSet = new UnionFindSet(n);
        for(int i=0;i<n;i++) {
            for(int j=0;j<n;j++) {
                if(isConnected[i][j]==1) {
                    unionFindSet.union(i,j);
                }
            }
        }
        return unionFindSet.getCount();
    }
}
 class UnionFindSet {
    public int[] elem;
    public int usedSize;

    public UnionFindSet(int n) {
        this.elem = new int[n];
        Arrays.fill(elem,-1);
    }

    /**
     * 查找数据x的根节点
     * @param x
     * @return
     */
    public int findRoot(int x) {
        if(x<0) {
            throw new IndexOutOfBoundsException("下标不合法，是负数");
        }
        while(elem[x]>=0) {
            x = elem[x];
        }
        return x;
    }

    /**
     * 判断x1和x2是不是同一个集合
     * @param x1
     * @param x2
     * @return
     */
    public boolean isSameUnionFindSet(int x1,int x2) {
        int index1= findRoot(x1);
        int index2 = findRoot(x2);
        return index1==index2;
    }

    /**
     * 合并操作
     * @param x1
     * @param x2
     */
    public void union(int x1,int x2) {
        int index1 = findRoot(x1);
        int index2 = findRoot(x2);
        if(index1==index2) {
            return;
        }
        elem[index1] = elem[index1]+elem[index2];
        elem[index2] = index1;
    }

    /**
     *
     * @return
     */
    public int getCount() {
        int count =0;
        for(int x:elem) {
            if(x<0) {
                count++;
            }
        }
        return count;
    }
    
}
